package com.wxapp.common.util.stream;

import com.wxapp.common.util.check.CheckUtil;

import java.util.Comparator;
import java.util.List;
import java.util.function.Function;
import java.util.function.Predicate;
import java.util.stream.Collectors;

/**
 * JDK8 stream流操作工具类
 */
public class StreamUtils {
    private StreamUtils(){}

    /**
     * 映射
     * @param data 不能为空
     * @param fun
     * @param <T>
     * @param <R>
     * @return data为空抛出异常IllegalArgumentException
     */
    public static <T,R> List<R> map(List<T> data, Function<T,R> fun){
        if (CheckUtil.isNullList(data)){
            throw new IllegalArgumentException();
        }
        return data.stream().map(fun).collect(Collectors.toList());
    }

    /**
     * 过滤
     * @param data
     * @param pre
     * @param <T>
     * @return data为空返回data
     */
    public static <T> List<T> filter(List<T> data, Predicate<T> pre){
        if (CheckUtil.isNullList(data)){
            return data;
        }
        return data.stream().filter(pre).collect(Collectors.toList());
    }

    /**
     * 排序
     * @param data
     * @param comparator
     * @param <T>
     * @return data为空返回data
     */
    public static <T> List<T> sorted(List<T> data, Comparator<T> comparator){
        if (CheckUtil.isNullList(data)){
            return data;
        }
        return data.stream().sorted(comparator).collect(Collectors.toList());
    }

    /**
     * 去重
     * @param data
     * @param <T>
     * @return data为空返回data
     */
    public static <T> List<T> distinct(List<T> data){
        if (CheckUtil.isNullList(data)){
            return data;
        }
        return data.stream().distinct().collect(Collectors.toList());
    }

    /**
     * 判断是否包含匹配元素
     * @param data
     * @param pre
     * @param <T>
     * @return data为空返回false
     */
    public static <T> boolean anyMatch(List<T> data, Predicate<T> pre){
        if (CheckUtil.isNullList(data)){
            return false;
        }
        return data.parallelStream().anyMatch(pre);
    }

    /**
     * 将list进行join操作
     * @param data
     * @param join
     * @return 返回join之后的字符串,data为空返回null
     */
    public static String join(List<String> data,String join){
        if (CheckUtil.isNullList(data)){
            return null;
        }
        return data.stream().collect(Collectors.joining(join == null ? "" : join));
    }
}